/**
 *
 * Copyright (c) 2006-2007 Berlin Brown and botnode.com/Berlin Research  All Rights Reserved
 *
 * http://www.opensource.org/licenses/bsd-license.php

 * All rights reserved.

 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:

 * * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * * Neither the name of the Botnode.com (Berlin Brown) nor
 * the names of its contributors may be used to endorse or promote
 * products derived from this software without specific prior written permission.

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Date: 12/15/2009 
 *   
 * Home Page: http://botnode.com/
 * 
 * Contact: Berlin Brown <berlin dot brown at gmail.com>
 * 
 * Also see: http://htmlcleaner.sourceforge.net/doc/org/htmlcleaner/class-use/TagNode.html
 */
package org.bresearch.octane.parse;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.bresearch.octane.parse.html.HtmlLink;
import org.htmlcleaner.CleanerProperties;
import org.htmlcleaner.HtmlCleaner;
import org.htmlcleaner.TagNode;

public class ParseHtml {

    private final HtmlCleaner htmlCleaner;
    private final String htmlData;
    private String currentURL;
    private String renderHtmlData = "";
    
    private List<HtmlLink> listLinks = new ArrayList<HtmlLink>();
    private int linksCount = 0;        
            
    public ParseHtml(final String currentURL, final HtmlCleaner cleaner, final String htmlData) {
        
        this.htmlCleaner = cleaner;
        this.htmlData = htmlData;
        this.currentURL = currentURL;
        
    }
    
    public String toString() {
        return "ParseHtml links=" + this.linksCount;
    }

    public void addLink(final String text, final String href) {
        this.listLinks.add(new HtmlLink(text, href));
    }
    
    public void buildLinks(final TagNode parentNode) {
        
        // optionally find parts of the DOM or modify some nodes
        final ValidLinkBuilder builder = new ValidLinkBuilder(this.currentURL, parentNode);
        builder.build();
        this.listLinks = builder.getListLinks();
        this.linksCount = this.listLinks.size();
    }
    
    public void parse() {       
        
        // take default cleaner properties
        CleanerProperties props = htmlCleaner.getProperties();
         
        // customize cleaner's behaviour with property setters
        //props.setAllowHtmlInsideAttributes();
         
        // Clean HTML taken from simple string, file, URL, input stream, 
        // input source or reader. Result is root node of created 
        // tree-like structure. Single cleaner instance may be safely used
        // multiple times.        
        try {
            
            final TagNode node = this.htmlCleaner.clean(this.htmlData);
            
            // Api: TagNode.getAllElements(boolean isRecursive) //
            final TagNode [] allNodes = node.getAllElements(true);
                    
            final StringBuilder htmlBuf = new StringBuilder(80);
            for (TagNode locNode : allNodes) {
                final String htmlText = locNode.getText().toString();
                if (htmlText.trim().length() > 0) {
                    htmlBuf.append(htmlText.trim());                    
                }
            }
            this.renderHtmlData = htmlBuf.toString();
            buildLinks(node);
            
        } catch(IOException ioe) {
            
            ioe.printStackTrace();
            
        } // End of the Try - Catch //
    }

    public String formatLinks() {
        
        // Iterate through the links
        // and store as a string.
        int i = 0;
        final StringBuilder buf = new StringBuilder(40);
        for (HtmlLink link : listLinks) {
            buf.append("link: " + i + ": " + link);
            buf.append('\n');
            i++;
        }
        return buf.toString();
    }
    
    /**
     * @return the listLinks
     */
    public List<HtmlLink> getListLinks() {
        return listLinks;
    }

    public HtmlLink getHtmlLink(final int id) {
        return this.listLinks.get(id);
    }

    
    /**
     * @param listLinks the listLinks to set
     */
    public void setListLinks(List<HtmlLink> listLinks) {
        this.listLinks = listLinks;
    }

    /**
     * @return the linksCount
     */
    public int getLinksCount() {
        return linksCount;
    }

    /**
     * @return the renderHtmlData
     */
    public String getRenderHtmlData() {
        return renderHtmlData;
    }

    /**
     * @param renderHtmlData the renderHtmlData to set
     */
    public void setRenderHtmlData(String renderHtmlData) {
        this.renderHtmlData = renderHtmlData;
    }

    /**
     * @return the currentURL
     */
    public String getCurrentURL() {
        return currentURL;
    }

    /**
     * @param currentURL the currentURL to set
     */
    public void setCurrentURL(String currentURL) {
        this.currentURL = currentURL;
    }
    
} // End of the Class //
